from PhotozSN import *

# making smaller mock SNe samples
PSN_main = PhotozSN(SN_want=400000, \
                    sigma_int=0.13, \
                    zp_min=0.1, zp_max=1.0, z_sn_max=2.0, \
                    n_mock=100, \
                    H0=67.7, Omega_m=0.307, \
                    Omega_k=0.0, w0=-1.0, wa=0.0, grid_size=100,\
                    sn_density="SN_density_lsst_SNR15_Nfilt3_main.txt",\
                    out_dir="SNR15_Nfilt3_main_400000" )
                    
PSN_main.make_sample()


## assuming 1200 x 10 deep filed SNe
#PSN_deep = PhotozSN(SN_want=12000, \
#                    sigma_int=0.13, \
#                    zp_min=0.15, zp_max=1.1, z_sn_max=2.0, \
#                    n_mock=100, \
#                    H0=67.7, Omega_m=0.307, \
#                    Omega_k=0.0, w0=-1.0, wa=0.0, grid_size=100,\
#                    sn_density="SN_density_lsst_SNR15_Nfilt3_deep.txt",
#                    out_dir="SNR15_Nfilt3_deep" )

#PSN_deep.make_sample()

## assuming 35000 x 10 main filed SNe
#PSN_main = PhotozSN(SN_want=350000, \
#                    sigma_int=0.13, \
#                    zp_min=0.1, zp_max=1.0, z_sn_max=2.0, \
#                    n_mock=100, \
#                    H0=67.7, Omega_m=0.307, \
#                    Omega_k=0.0, w0=-1.0, wa=0.0, grid_size=100,\
#                    sn_density="SN_density_lsst_SNR15_Nfilt3_main.txt",\
#                    out_dir="SNR15_Nfilt3_main" )
#                    
#PSN_main.make_sample()


